python square root of a number